package cn.tedu.v5;

import java.io.*;
import java.net.*;
import java.util.*;

/**
 * Version 5 多人聊天+QQ聊天+消息转发
 * 功能： 登录 + 聊天 【退出】
 * 登录步骤：1.私有化属性，保存账号与密码--userDatabase
 *         2.Server构造器，创建两个用户名与密码
 *         3.处理多个客户端，加私有化属性：账号、密码
 */
public class Server {
    private ServerSocket serverSocket;
    private static final Integer PORT=5555;
    private List<PrintWriter> allOut=new ArrayList<>();
    //1.私有化属性--保存账号与密码--onlineUsers
    private Set<String> onlineUsers=new HashSet<>();
    private Map<String,String> userDatabase;

    public Server() {
        try {
            serverSocket = new ServerSocket(PORT);
            //2.Server对象启动--main方法--
            // 创建服务器之初，完成对象账号与密码赋值
            userDatabase=new HashMap<>();
            userDatabase.put("delight","5555");
            userDatabase.put("happy","6666");
        } catch (BindException e) {
            System.out.println("端口号重复申请，建议换个端口号");
        } catch (IOException e) {
            System.out.println("申请端口号失败....");
        }
    }

    public void start() {
        try {
            while (true) {
                System.out.println("----等待客户端连接----");
                Socket socket = serverSocket.accept();
                new Thread(new ClientHandler(socket)).start();
            }
        } catch (IOException e) {
            System.out.println("服务器端通讯断开");
        } finally {
            if (serverSocket != null && !serverSocket.isClosed()) {
                try {
                    serverSocket.close();
                } catch (IOException e) {
                    System.out.println("服务器关闭失败");
                }
            }
        }
    }

    public static void main(String[] args) {
        new Server().start();
    }
    private synchronized void broadcastMessage(String message){
        for (PrintWriter pw:allOut){
            pw.println("消息广播："+message);
        }
    }
    //2.封装方法：同步转发给所有客户端消息："在线用户："+onlineUsers
    //updateOnlineUsers 更新在线人名单的方法："在线用户："+onlineUsers
    //字符串：String userList="在线用户"+String.join("、",onlineUsers)
    //userList写出给所有客户端
    private synchronized void updateOnlineUsers(){
        String userList="在线用户："+String.join("、",onlineUsers);
        for (PrintWriter pw:allOut){
            pw.println(userList);
        }
    }

    private class ClientHandler implements Runnable {
        private Socket socket;
        private String ip;
        private BufferedReader br;
        private PrintWriter pw;
        //3.很多客户端都具备账号密码
        private String username;
        private String password;

        public ClientHandler(Socket s) {
            this.socket = s;
            ip = socket.getInetAddress().getHostAddress();
            try {
                br = new BufferedReader(new InputStreamReader(socket.getInputStream(), "UTF8"));
                pw = new PrintWriter(new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(), "UTF8")), true);
                synchronized (allOut) {
                    allOut.add(pw);
                }

            } catch (IOException e) {
                System.out.println("客户端断开通讯");
            }
        }

        //4.4.1 封装判断账号与密码是否正确的方法--Login
        private boolean Login(String username,String password){
            //Map中是否存在这个账号和密码
            //userDatabase.containsKey(username) 判断账号存在
            //userDatabase.get(username).equals(password) 判断这个账号密码是否正确
            return userDatabase.containsKey(username)&&userDatabase.get(username).equals(password);
        }
        //5.账号与密码验证成功---登录---聊天：读写功能--封装方法handlerChat()
        private void handlerChat(){
            new Thread(() -> {
                try {
                    String message;
                    while ((message = br.readLine()) != null) {
                        System.out.println("客户端说：" + message);
                        System.out.println("服务器端请输入：");
                        broadcastMessage(message);
                    }
                } catch (IOException e) {
                    System.out.println("客户端读入失败");
                }
            }).start();
            Scanner input = new Scanner(System.in);
            while (true) {
                System.out.println("服务器请输入：");
                String say = input.nextLine();
                //pw.println("服务器说：" + say);
                //1.如果服务器写出"exit"，断开连接
                if ("exit".equalsIgnoreCase(say)){
                    pw.println("exit");
                    System.out.println("happy客户端断开连接：IP是："+ip);
                    break;
                }else {
                    //2.如果没有输入"exit"则继续输入
                    pw.println("服务器请输入："+say);
                }
            }
        }

        public void run() {
            try {
                //4.目的：启动客户端与服务器端连接
                //  假设：一个客户端连接上了，服务器端应该怎么处理
                //4.1  登录验证 【服务器端与客户端交互问题】
                pw.println("请输入用户名：");//发给客户端
                username=br.readLine();//读入客户端发来的账号
                //4.2  密码验证
                pw.println("请输入密码：");//发给客户端
                password=br.readLine();//读入客户端发来的密码
                //4.3  判断账号密码是否正确---万一账号很多情况
                //4.4  账号很多情况：封装方法--功能：判断账号密码是否正确
                if (Login(username,password)){//判断Login为true 账号与密码正确
                    pw.println("登录成功，欢迎进入happy牌聊天室");
                    //3.所有用户添加至集合中【同步块：登录集合】
                    //3.1转发在线用户方法
                    synchronized (onlineUsers){
                        onlineUsers.add(username);
                        updateOnlineUsers();
                    }
                    //聊天 4.5封装方法--登录成功--开始聊天：读写功能--handlerChat
                    handlerChat();
                }else{//不通过
                    pw.println("登录失败，用户名或密码错误");
                    socket.close();//登录失败，关闭连接
                }
            } catch (NullPointerException e){
                System.out.println("IO流传入对象，" +
                        "有对象为null，立马核对，" +
                        "建议：查看try块中对象可能赋值为null");
            } catch (IOException e) {
                System.out.println("服务器端断开连接");
            } finally {
                //4.客户端断开连接，移除当前集合【同步】
                synchronized (onlineUsers) {
                    allOut.remove(onlineUsers);
                }
                synchronized (allOut){
                    allOut.remove(pw);
                }
                if (br != null)
                    try {
                        br.close();
                    } catch (IOException e) {
                        System.out.println("结束通讯：关闭失败");
                    }
                if (pw!=null) pw.close();
                if (socket != null && !socket.isClosed())
                    try {
                        socket.close();
                    } catch (IOException e) {
                        System.out.println("通讯关闭失败");
                    }
            }
        }
    }
}

